package Leetcode.DayTest;

import java.util.*;

/**
 * @Author: kirito
 * @Date: 2024/5/23 13:00
 * @Description:
 * 2831. 找出最长等值子数组
 * 中等
 * 相关标签
 * 相关企业
 * 提示
 * 给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。
 *
 * 如果子数组中所有元素都相等，则认为子数组是一个 等值子数组 。注意，空数组是 等值子数组 。
 *
 * 从 nums 中删除最多 k 个元素后，返回可能的最长等值子数组的长度。
 *
 * 子数组 是数组中一个连续且可能为空的元素序列。
 *
 *
 *
 * 示例 1：
 *
 * 输入：nums = [1,3,2,3,1,3], k = 3
 * 输出：3
 * 解释：最优的方案是删除下标 2 和下标 4 的元素。
 * 删除后，nums 等于 [1, 3, 3, 3] 。
 * 最长等值子数组从 i = 1 开始到 j = 3 结束，长度等于 3 。
 * 可以证明无法创建更长的等值子数组。
 * 示例 2：
 *
 * 输入：nums = [1,1,2,2,1,1], k = 2
 * 输出：4
 * 解释：最优的方案是删除下标 2 和下标 3 的元素。
 * 删除后，nums 等于 [1, 1, 1, 1] 。
 * 数组自身就是等值子数组，长度等于 4 。
 * 可以证明无法创建更长的等值子数组。
 */

public class longestEqualSubarray {
    public int longestEqualSubarray(List<Integer> nums, int k) {
        // 计算数组的长度
        int n = nums.size();
        // 创建一个二维数组，用于存储每个数字在数组中的位置列表
        List<Integer>[] posLists = new ArrayList[n + 1];
        // 使用Arrays.setAll()方法初始化所有的位置列表
        Arrays.setAll(posLists, i -> new ArrayList<>());
        // 遍历数组
        for (int i = 0; i < n; i++) {
            // 获取当前元素
            int x = nums.get(i);
            // 将当前元素的位置添加到对应数字的位置列表中
            posLists[x].add(i - posLists[x].size());
        }

        // 初始化最长子数组的长度为0
        int ans = 0;
        // 遍历所有的位置列表
        for (List<Integer> pos : posLists) {
            // 如果当前列表的长度小于或等于已知的最大长度，则跳过
            if (pos.size() <= ans) {
                continue; // 无法让 ans 变得更大
            }
            // 初始化左侧指针
            int left = 0;
            // 遍历当前列表
            for (int right = 0; right < pos.size(); right++) {
                // 如果当前元素与左侧指针指向的元素之间的距离大于k，则移动左侧指针
                while (pos.get(right) - pos.get(left) > k) { // 要删除的数太多了
                    left++;
                }
                // 更新最长子数组的长度
                ans = Math.max(ans, right - left + 1);
            }
        }
        // 返回最长子数组的长度
        return ans;
//        int n=nums.size();
//        int[] nc=new int[n+1];
//        int mc=0,i=0,j=0;
//        while(i<n){
//            int num=nums.get(i++);
//            nc[num]++;
//            mc=Math.max(nc[num],mc);
//            if(mc+k<i-j)nc[nums.get(j++)]--;
//        }
//        return mc;
    }


}
